/*
 * Copyright (C) 2011-2021 Intel Corporation. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 *   * Redistributions of source code must retain the above copyright
 *     notice, this list of conditions and the following disclaimer.
 *   * Redistributions in binary form must reproduce the above copyright
 *     notice, this list of conditions and the following disclaimer in
 *     the documentation and/or other materials provided with the
 *     distribution.
 *   * Neither the name of Intel Corporation nor the names of its
 *     contributors may be used to endorse or promote products derived
 *     from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 */

#ifndef _ENCLAVE_H_
#define _ENCLAVE_H_

#include <assert.h>
#include <stdlib.h>
#include <sgx_trts.h>
#include <sgx_tcrypto.h>

#if defined(__cplusplus)
extern "C" {
#endif

int printf(const char* fmt, ...);
// void enclave_increment_array(int array[5], int len);
// void ecall_sum(int array[1000000], int len);
int cal_prime();
// void enclave_main(char (*lines)[5], int line_count);
// void xor_block(__uint8_t *a, __uint8_t *b, __uint8_t *out);
// void shift_left(__uint8_t *input);
// void generate_subkeys(__uint8_t *key, __uint8_t *k1, __uint8_t *k2);
// void aes_cmac(__uint8_t *key, __uint8_t *data, size_t size, __uint8_t *mac);
// void encrypt_data(const __uint8_t* input, size_t input_size, __uint8_t* key, __uint8_t* iv, __uint8_t* output, size_t output_size);
// void compute_aes_cmac(const __uint8_t* input, size_t input_size, __uint8_t* key, __uint8_t* cmac);
// void xor_block(__uint8_t* block1, const __uint8_t* block2);
// sgx_status_t sgx_init_crypto_lib(void);
// sgx_status_t sgx_rijndael128_cmac_msg(const sgx_cmac_128bit_key_t *p_key,
//                                     const uint8_t *p_src,
//                                     uint32_t src_len,
//                                     sgx_cmac_128bit_tag_t *p_mac);
// void sgx_destroy_crypto_op(void);
sgx_status_t calculate_cmac(const uint8_t* message, size_t message_len, sgx_cmac_128bit_tag_t* cmac);

#if defined(__cplusplus)
}
#endif

#endif /* !_ENCLAVE_H_ */
